#pragma once

#include <memory>
#include "Arguments.h"
#include "Children.h"
#include "Context.h"
#include "SourceFile.h"

class UserModule;

class ScopeContext : public Context
{
public:
  void init() override;
  boost::optional<CallableFunction> lookup_local_function(const std::string& name, const Location& loc) const override;
  boost::optional<InstantiableModule> lookup_local_module(const std::string& name, const Location& loc) const override;

protected:
  ScopeContext(const std::shared_ptr<const Context>& parent, const LocalScope *scope) :
    Context(parent),
    scope(scope)
  {}

private:
// Experimental code. See issue #399
//	void evaluateAssignments(const AssignmentList &assignments);

  const LocalScope *scope;

  friend class Context;
};

class UserModuleContext : public ScopeContext
{
public:
  const Children *user_module_children() const override { return &children; }
  std::vector<const std::shared_ptr<const Context> *> list_referenced_contexts() const override;

protected:
  UserModuleContext(const std::shared_ptr<const Context>& parent, const UserModule *module, const Location& loc, Arguments arguments, Children children);

private:
  Children children;

  friend class Context;
};

class FileContext : public ScopeContext
{
public:
  boost::optional<CallableFunction> lookup_local_function(const std::string& name, const Location& loc) const override;
  boost::optional<InstantiableModule> lookup_local_module(const std::string& name, const Location& loc) const override;

protected:
  FileContext(const std::shared_ptr<const Context>& parent, const SourceFile *source_file) :
    ScopeContext(parent, &source_file->scope),
    source_file(source_file)
  {}

private:
  const SourceFile *source_file;

  friend class Context;
};
